from php4dvd.pages.blocks.film_form import FilmForm
from php4dvd.pages.internal_page import InternalPage
from selenium.webdriver.common.by import By


class AddFilmPage(InternalPage):

    @property
    def save_button(self):
        return self.driver.find_element_by_css_selector("img[alt=\"Save\"]")


    def __init__(self, driver, base_url):
        super(AddFilmPage, self).__init__(driver, base_url)
        self.film_form = FilmForm(self.driver, self.base_url)
